package com.bdqn;

import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Text;
import org.xml.sax.SAXException;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.*;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import java.io.FileOutputStream;
import java.io.IOException;

/**
 * 使用DOM维护XML数据
 */
public class Demo03 {

    public static void main(String[] args) {
        // 得到DOM解析器的工厂实例
        DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
        try {
            // 从DOM工厂获得DOM解析器
            DocumentBuilder db = dbf.newDocumentBuilder();
            // 解析XML文档，得到一个Document对象，即DOM树
            Document doc = db.parse("ch02/src/main/resources/维护收藏信息.xml");

            // -- start 添加节点
            Text entNode = doc.createTextNode("\n");
            Text tabNode = doc.createTextNode("\t");

            // 创建 Brand 元素节点
            Element brandElement = doc.createElement("Brand");
            // 为 Brand 节点设置属性
            brandElement.setAttribute("name", "三星");

            // 创建 Type 元素节点
            Element typeElement = doc.createElement("Type");
            typeElement.setAttribute("name", "NoteX");

            // 添加父子关系 将Type作为子节点添加到Brand节点中
            brandElement.appendChild(entNode);
            brandElement.appendChild(tabNode);
            brandElement.appendChild(tabNode);
            brandElement.appendChild(typeElement);
            brandElement.appendChild(entNode);
            brandElement.appendChild(tabNode);

            // 获得根节点元素 根节点只有一个，所以用 item(0) 直接获取元素
            Element phoneElement = (Element) doc.getElementsByTagName("PhoneInfo").item(0);
            // 将Brand作为子节点添加到根节点PhoneInfo中
            phoneElement.appendChild(tabNode);
            phoneElement.appendChild(brandElement);
            // phoneElement.appendChild(entNode);

            System.out.println("Append Child OK");
            // -- end 添加节点

            // -- start 修改节点
            /* NodeList listEdit = doc.getElementsByTagName("Brand");
            for (int i = 0; i < listEdit.getLength(); i++) {
                Element brandEle = (Element) listEdit.item(i);
                // 获取元素节点的name属性
                String brandName = brandEle.getAttribute("name");
                // 找到name="三星"的元素节点
                if (brandName.equals("三星")) {
                    // 修改元素节点的属性
                    brandEle.setAttribute("name", "SAMSUNG");
                }
            }

            System.out.println("Edit Child OK"); */
            // -- end 修改节点

            // -- start 删除节点
            /* NodeList listRemove = doc.getElementsByTagName("Brand");
            for (int i = 0; i < listRemove.getLength(); i++) {
                Element brandEle = (Element) listRemove.item(i);
                // 获取元素节点的name属性
                String brandName = brandEle.getAttribute("name");
                // 找到name="三星"的元素节点
                if (brandName.equals("苹果")) {
                    // 删除元素节点
                    brandEle.getParentNode().removeChild(brandEle);
                }
            }

            System.out.println("Remove Child OK"); */
            // -- end 删除节点

            // 保存XML文件
            TransformerFactory transformerFactory = TransformerFactory.newInstance();
            Transformer transformer = transformerFactory.newTransformer();
            DOMSource domSource = new DOMSource(doc);
            // 设置编码
            transformer.setOutputProperty(OutputKeys.ENCODING, "UTF-8");
            StreamResult result = new StreamResult(new FileOutputStream("ch02/src/main/resources/维护收藏信息.xml"));
            // 把DOM树转换为XML文件
            transformer.transform(domSource, result);
            System.out.println("XML Saved");


        } catch (ParserConfigurationException e) {
            e.printStackTrace();
        } catch (SAXException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (TransformerConfigurationException e) {
            e.printStackTrace();
        } catch (TransformerException e) {
            e.printStackTrace();
        }
    }
}
